from phi.assistant import Assistant
from phi.llm.deepseek import DeepSeek

# 创建 DeepSeek LLM
llm = DeepSeek(
    api_key="sk-e018d19a2b36475aa29e6104fe91fac9",
    model="deepseek-chat",
)

# 创建 Assistant，将 role 改为 "system" 或 "user"
agent = Assistant(
    llm=llm,
    system_prompt="You are a helpful AI assistant.",  # 使用 system_prompt 而不是 role
)

agent.print_response("Share a 2 sentence horror story")
